package com.five.base.configuration;

import com.five.base.common.BaseEmail;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;

import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.mail.*;
import javax.mail.internet.*;
import java.io.UnsupportedEncodingException;
import java.util.Date;
import java.util.Properties;

/**
 * @ClassName MailConfiguration
 * @Author wuxuan
 * @Date 2019/1/6 16:56
 * @Version 1.0
 * @Description 邮件配置类
 **/
@Configuration
@Component
@Slf4j
public class MailConfiguration extends BaseEmail {

    @Value("${base.localFileAddress}")
    String localFile;

    @Override
    public boolean sendMail() {
        log.info("-------开始发送邮件--------");
        boolean isSuccess = true;
        //1、连接邮件服务器的参数配置
        try {
            Properties props = new Properties();
            //设置用户的认证方式
            props.setProperty("mail.smtp.auth", "true");
            //设置传输协议
            props.setProperty("mail.transport.protocol", MAIL_PROTOCOL);
            //设置发件人的SMTP服务器地址
            props.setProperty("mail.smtp.host", MAIL_HOST);
            //2、创建定义整个应用程序所需的环境信息的 Session 对象
            Session session = Session.getInstance(props);
            //设置调试信息在控制台打印出来
            session.setDebug(false);
            //3、创建邮件的实例对象
            Message msg = getMimeMessage(session);
            //4、根据session对象获取邮件传输对象Transport
            Transport transport = session.getTransport();
            //设置发件人的账户名和密码
            transport.connect(MAIL_SENDER, MAIL_CODE);
            //发送邮件，并发送到所有收件人地址，message.getAllRecipients() 获取到的是在创建邮件对象时添加的所有收件人, 抄送人, 密送人
            transport.sendMessage(msg, msg.getAllRecipients());
            //5、关闭邮件连接
            transport.close();
        } catch (MessagingException e) {
            e.printStackTrace();
            isSuccess = false;
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
            isSuccess = false;
        }
        if (isSuccess){
            log.info("---------邮件发送成功----------");
        }else {
            log.info("-----------邮件发送失败----------");
        }
        return isSuccess;
    }

    private Message getMimeMessage(Session session) throws MessagingException, UnsupportedEncodingException, javax.mail.MessagingException {
        //创建一封邮件的实例对象
        MimeMessage msg = new MimeMessage(session);
        //设置发件人地址
        msg.setFrom(new InternetAddress(MAIL_SENDER));
        /**
         * 设置收件人地址（可以增加多个收件人、抄送、密送），即下面这一行代码书写多行
         * MimeMessage.RecipientType.TO:发送
         * MimeMessage.RecipientType.CC：抄送
         * MimeMessage.RecipientType.BCC：密送
         */
        msg.setRecipient(MimeMessage.RecipientType.TO, new InternetAddress(RECEIVER));
        //设置邮件主题
        msg.setSubject(MAIL_SUBJECT, "UTF-8");
        //添加附件节点
        MimeBodyPart part = new MimeBodyPart();
        // 读取本地文件
        DataHandler dh2 = new DataHandler(new FileDataSource(localFile));
        // 将附件数据添加到"节点"
        part.setDataHandler(dh2);
        // 设置附件的文件名（需要编码）
        part.setFileName(MimeUtility.encodeText(dh2.getName()));
        //添加文本节点
        MimeBodyPart text = new MimeBodyPart();
        text.setContent(MAIL_CONTEXT,"text/html;charset=UTF-8");

        MimeMultipart mime = new MimeMultipart();
        // 如果有多个附件，可以创建多个多次添加
        mime.addBodyPart(part);
        mime.addBodyPart(text);
        // 混合关系
        mime.setSubType("mixed");
        //设置邮件正文
        msg.setContent(mime);
        //设置邮件的发送时间,默认立即发送
        msg.setSentDate(new Date());
        return msg;
    }


}
